﻿// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System.Text;
using System.Text.RegularExpressions;

namespace Microsoft.Extensions.AI.Templates.Tests;

internal static class VerifyScrubbers
{
    /// <summary>
    /// Removes content after "Details: ".
    /// </summary>
    internal static void ScrubDetails(this StringBuilder output)
    {
        output.ScrubByRegex("(Details: )([^\\r\\n]*)", $"Details: %DETAILS%");
    }

    /// <summary>
    /// Removes table header delimiter.
    /// </summary>
    internal static void ScrubTableHeaderDelimiter(this StringBuilder output)
    {
        output.ScrubByRegex("---[- ]*", "%TABLE HEADER DELIMITER%");
    }

    /// <summary>
    /// Replaces Windows newlines (CRLF) with Unix style newlines (LF).
    /// </summary>
    /// <param name="output"></param>
    internal static StringBuilder UnixifyNewlines(this StringBuilder output)
    {
        return output.Replace("\r\n", "\n");
    }

    /// <summary>
    /// Replaces Windows Directory separator char (\) with Unix Directory separator char (/).
    /// </summary>
    /// <param name="output"></param>
    internal static string UnixifyDirSeparators(this string output)
    {
        return output.Replace('\\', '/');
    }

    /// <summary>
    /// Replaces content matching <paramref name="pattern"/> with <paramref name="replacement"/>.
    /// </summary>
    internal static void ScrubByRegex(this StringBuilder output, string pattern, string replacement, RegexOptions regexOptions = RegexOptions.None)
    {
        string finalOutput = Regex.Replace(output.ToString(), pattern, replacement, regexOptions);
        output.Clear();
        output.Append(finalOutput);
    }

    /// <summary>
    /// Replaces content matching <paramref name="textToReplace"/> with <paramref name="replacement"/>.
    /// </summary>
    internal static void ScrubAndReplace(this StringBuilder output, string textToReplace, string replacement)
    {
        string finalOutput = output.ToString().Replace(textToReplace, replacement);
        output.Clear();
        output.Append(finalOutput);
    }
}
